// 该案例是统计目录内的文件
// 通过opendir和readdir函数对制定目录进行遍历操作
// 然后打印输出到指定目录中各种类型的文件的相应数据实例
// 制定目录的名称通过argv参数给出

#include <stdio.h>
#include <fcntl.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <dirent.h>
#include <string.h>
#include <stdlib.h>


// 函数定义
typedef int Myfunc(const char*, const struct stat *, int);
static Myfunc myfunc;

static int myftw(char*, Myfunc *);

static int dopath(Myfunc *);

// 各种类型的文件数目对应的变量
static long nreg, ndir, nblk, nchr, nfifo, nslink, nsock, ntot;

char *path_alloc(int *size);


int main(int argc, char *argv[]){
	
	int ret;

	if(argc != 2){
		printf("请输入正确的参数！\n");
		return -2;
	}
	
	/*does it all*/
	ret = myftw(argv[1],myfunc);

	// 计算文件总量
	ntot = nreg + ndir + nchr + nfifo + nslink + nsock;
	
	// 如果目录中没有文件，则讲ntot设置为1以避免除数为0
	if(ntot == 0){
		ntot = 1;
	}

	// 以下为一次打印各种类型文件的数据
	printf("普通文件 = %7ld, %5.2f%%\n", nreg, nreg*100.0/ntot);
	printf("目录文件 = %7ld, %5.2f%%\n", ndir, ndir*100.0/ntot);
	printf("块设备文件 = %7ld, %5.2f%%\n", nblk, nblk*100.0/ntot);
	printf("字设备文件 = %7ld, %5.2f%%\n", nchr, nchr*100.0/ntot);
	printf("FIFOs = %7ld, %5.2f%%\n", nfifo, nfifo*100.0/ntot);

	printf("链接符号文件 = %7ld, %5.2f%%\n", nslink, nslink*100.0/ntot);
	printf("套接字文件 = %7ld, %5.2f%%\n", nsock, nsock*100.0/ntot);

	return ret;
}

// 路径缓冲区分配函数
char *path_alloc(int *size){
	
	char *p = NULL;

	// 如果size为0，则退出该函数并返回NULL
	if(!size){
		return NULL;
	}
	
	// 如果size不为0,则分配内存
	p = malloc(256);

	if(p){
		*size = 256;
	}else{
		*size = 0;
	}

	return p;

}

#define FTW_F	1	//
#define FTW_D	2	// 目录
#define FTW_DNR	3	// 不可读的目录
#define FTW_NS	4	// 不可获取状态的文件


// 存放每个文件的完整路径
static char *fullpath;

static int myftw(char *pathname, Myfunc *func){
	int len;
	// 给路径缓冲区分配一个长度
	fullpath = path_alloc(&len);
	// 复制文件名称
	strncpy(fullpath, pathname, len);

	fullpath[len-1]=0;
	return(dopath(func));
}

// 获得文件的状态
static int dopath(Myfunc* func){

	struct stat statbuf;
	struct dirent *dirp;
	DIR *dp;
	int ret;
	char *ptr;

	// 获得文件状态失败
	if(lstat(fullpath,&statbuf)<0){
		return(func(fullpath, &statbuf, FTW_NS));
	}

	// 如果不是目录
	if(S_ISDIR(statbuf.st_mode)==0){
		return(func(fullpath, &statbuf, FTW_F));
	}

	if(ret=func(fullpath, &statbuf, FTW_D) != 0){
		return(ret);
	}
	
	// 指向路径缓冲区结尾
	ptr = fullpath + strlen(fullpath);

	*ptr++ = '/';
	*ptr = 0;

	// 如果不能读取目录
	if((dp = opendir(fullpath)) == NULL){
		return(func(fullpath, &statbuf, FTW_DNR));
	}

	while((dirp = readdir(dp)) != NULL){
		if(strcmp(dirp->d_name, ".") == 0 || strcmp(dirp->d_name, ".." == 0))
			continue;		/*ignore dot and dot-dot*/

		strcpy(ptr, dirp->d_name);	/*append name after slash*/

		if((ret = dopath(func)) != 0)	/*recursive*/
			break;			/*time to leave*/

	}

	/*erase everything from slash onwards*/
	ptr[-1] = 0;

	if(closedir(dp) < 0){
		printf("can't close directory %s\n", fullpath);
	}

	return(ret);
}

static int myfunc(const char *pathname, const struct stat *statptr, int type){
	switch(type){
		case FTW_F:
			switch(statptr->st_mode & S_IFMT){
				case S_IFREG: nreg++;	break;
				case S_IFBLK: nblk++;	break;
				case S_IFCHR: nchr++;	break;
				case S_IFIFO: nfifo++;	break;
				case S_IFLNK: nslink++;	break;
				case S_IFSOCK: nsock++;	break;
				case S_IFDIR:
					       printf("for S_IFDIR for %s\n",pathname);
			}
			break;
		case FTW_D:
			ndir++;
			break;
		case FTW_DNR:
			printf("can't read directory %s\n", pathname);
			break;
		case FTW_NS:
			printf("stat error for %s\n", pathname);
			break;
		default:
			printf("unknown type %d for pathname %s\n", type, pathname);
	}

	return(0);
}


